Pytorch Data Augmentation Using Torchvision